package com.cn.wanxi.util;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.Date;
import java.util.Iterator;
import java.util.List;

public class LoadImg {
    /**
     * 图片上传
     *
     * @param req
     * @return 返回key值用于得到图片路劲，保存到数据库
     */
    public static String load(HttpServletRequest req, HttpServletResponse resp) throws IOException {
//        每一次文件上传的时候也需要判断是否已经登录
        String loginName = req.getParameter("loginName");
        String username = (String) req.getSession().getAttribute(loginName);
        if (username == null) {
            resp.sendRedirect("/html/login.html");
            return "";
        } else {
            String key = "";

//        判断是否是文件格式
            if (ServletFileUpload.isMultipartContent(req)) {

                try {
                    // 1. 创建DiskFileItemFactory对象，设置缓冲区大小和临时文件目录
                    DiskFileItemFactory factory = new DiskFileItemFactory();
                    // System.out.println(System.getProperty("java.io.tmpdir"));//默认临时文件夹

                    // 2. 创建ServletFileUpload对象，并设置上传文件的大小限制。
                    ServletFileUpload sfu = new ServletFileUpload(factory);
                    sfu.setSizeMax(10 * 1024 * 1024);// 以byte为单位 不能超过10M 1024byte =
                    // 1kb 1024kb=1M 1024M = 1G
                    sfu.setHeaderEncoding("utf-8");

                    // 3.
                    // 调用ServletFileUpload.parseRequest方法解析request对象，得到一个保存了所有上传内容的List对象。
                    @SuppressWarnings("unchecked")

                    List<FileItem> fileItemList = sfu.parseRequest(req);
                    Iterator<FileItem> fileItems = fileItemList.iterator();
//                List和Iterator的区别？
                    // 4. 遍历list，每迭代一个FileItem对象，调用其isFormField方法判断是否是上传文件
                    while (fileItems.hasNext()) {
//                    得到下一个元素对象
                        FileItem fileItem = fileItems.next();
                        // 普通表单元素
                        if (fileItem.isFormField()) {
                            String name = fileItem.getFieldName();// name属性值
                            String value = fileItem.getString("utf-8");// name对应的value值
                            req.setAttribute(name, value);
//                        String [] results="".split(",");
//                        CompanyModel companyModel=new CompanyMode();
//                        companyModel.setId(results[0]);
//                        companyModel.setName(results[1]);
//                        companyModel.setAddress(results[2]);
//                        System.out.println(name + " = " + value);
                        }
                        // <input type="file">的上传文件的元素
                        else {
                            key = fileItem.getFieldName();
                            String fileName = fileItem.getName();// 文件名称
                            //System.out.println("原文件名：" + fileName);// Koala.jpg

//                        String suffix = fileName.substring(fileName.lastIndexOf('.'));
                            //System.out.println("扩展名：" + suffix);// .jpg
//                        UUID.randomUUID().toString()//通用唯一标识符
                            // 新文件名（唯一）
                            String newFileName = new Date().getTime() + "---" + fileName;
                            //System.out.println("新文件名：" + newFileName);// image\1478509873038--a.jpg
//                        String user_img = "img/" + newFileName;//插入数据库url

//                        绝对路径
//                        String path = "D:\wanxi\canyin04-0119\web";
//                        String path = System.getProperty("user.dir");
                            String realPath = req.getSession().getServletContext().getRealPath("");
                            String path = realPath.split("out")[0] + "web";
                            String imgPath = "/loadImg/";//用户将图片上传放在一个固定的文件里面
//                        String fileName="";

                            // 5. 调用FileItem的write()方法，写入文件
                            File fileDerectory = new File(path + imgPath);//路径改成自己服务器存放图片的路径，如果是本机的直接写磁盘路径，远程服务器的话最好研究一下ftp的协议，这个方法一样，就是要多一个类进行远程接口调用
                            if (!fileDerectory.exists()) {
                                fileDerectory.mkdirs();
                            }
                            File file = new File(fileDerectory, newFileName);
//                        System.out.println(file.getAbsolutePath());
                            fileItem.write(file);
                            req.setAttribute(key, imgPath + newFileName);
                            System.out.println(file);
                            // 6. 调用FileItem的delete()方法，删除临时文件
                            fileItem.delete();
                            copy(file, newFileName, realPath + imgPath);
                        }
                    }

                } catch (FileUploadException e) {
                    e.printStackTrace();
                } catch (Exception e) {
                    e.printStackTrace();
                }

            }
            return key;
        }
    }

    /**
     * @param file     需要复制的文件
     * @param fileName 文件名
     */
    private static void copy(File file, String fileName, String path) {
        File fileDerectory = new File(path);
        if (!fileDerectory.exists()) {
            fileDerectory.mkdirs();
        }
//        将传递过来的文件复制到下面这个文件里面
        File f = new File(fileDerectory, fileName);//写在out里面
        try {
            InputStream inputStream = new FileInputStream(file);
            OutputStream outputStream = new FileOutputStream(f);
            int leng = 0;
            while ((leng = inputStream.read()) != -1) {
                outputStream.write(leng);
            }
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }

    }
}
